我有一个2D掩码数组,我想使用True值的逻辑OR操作沿着轴0折叠它。我想知道是否有一个numpy函数来完成这个过程。我的代码看起来像这样: >>> all_masks
array([[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, False, False, ..., False, False, False],
[False, True, False, ..., False, True, False],
[False, False, False, ..., False, False, False],
[False, True, False, ..., False, True, False]])
>>> all_masks.shape
(6, 870)
>>> output_mask
array([False, True, False, ..., False, True, False])
>>> output_mask.shape
(870,)我已经通过使用output_mask循环实现了这个过程。然而,我知道使用for循环会使我的代码变慢(并且有点乱),所以我想知道这个过程是否可以通过numpy或类似的函数来完成? 使用for循环折叠掩码的代码: mask_out = np.zeros(all_masks.shape[1], dtype=bool)
for mask in all_masks:
mask_out = mask_out | mask
return mask_out
|